
import torch
import numpy    as np
import torch.nn as nn

n, d = 8, 16
x = np.random.rand(n, d).astype(np.float16)
y = np.random.rand(n, d).astype(np.float16)

y = nn.functional.gelu(torch.from_numpy(x)).numpy()


x.tofile('../data/input_0.bin')
y.tofile('../data/output_0.bin')

